package com.chenwc.fileUtils;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author chenwc
 * @date 2024/4/1 22:23
 */
public class ReadFile {

    private static final Logger log = LoggerFactory.getLogger(ReadFile.class);


    /**
     * 以UTF-8字符编码读取文本文件内容（字符流）
     * @param path 文件路径
     * @return 文件内容
     */
    public static String readFileByInputStreamReader(String path) {
        return readFileByInputStreamReader(path, StandardCharsets.UTF_8);
    }

    /**
     * 以指定字符编码读取文本文件内容（字符流）
     * @param path 文件路径
     * @param charset	读取文件使用的字符编码
     * @return 文件内容
     */
    public static String readFileByInputStreamReader(String path, Charset charset) {
        File file = new File(path);
        log.info("读取文件路径: {}", file.getAbsolutePath());
        InputStreamReader inputStreamReader = null;
        BufferedReader br = null;
        StringBuilder sb = new StringBuilder();
        try {
            if (!file.exists()) {
                log.info("文件不存在，读取文件失败！");
                return null;
            }
            long start = System.currentTimeMillis();
            log.info("开始读取文件............");
            inputStreamReader = new InputStreamReader(new FileInputStream(file), charset);
            br = new BufferedReader(inputStreamReader);
            String string = null;
            //按行读取文件
            while ((string = br.readLine()) != null) {
                sb.append(string).append("\r\n");
            }
            long end = System.currentTimeMillis();
            log.info("读取文件成功............");
            log.info("读取文件耗时: {} ms", String.valueOf((end - start)));
        } catch (Exception e) {
            // TODO: handle exception
            e.printStackTrace();
        } finally {
            try {
                if (null != br) {
                    br.close();
                }
                if (null != inputStreamReader) {
                    inputStreamReader.close();
                }
            } catch (Exception e2) {
                // TODO: handle exception
                e2.printStackTrace();
            }
        }
        return sb.toString();
    }


    /**
     * 通过 FileInputStream 读取文件内容（字节流），最大读取文件大小为 Integer.MAX_VALUE (B)
     * @param fis FileInputStream
     * @return 二进制文件内容
     */
    public static byte[] readFileByInputStream(FileInputStream fis) {
        byte[] result = null;
        try {
            long start = System.currentTimeMillis();
            log.info("开始读取文件............");
            int available = fis.available();
            log.info("available: {}", String.valueOf(available));
            if (available <= 0) {
                log.info("待读取文件为空文件，读取文件失败");
                return null;
            }
            result = new byte[available];
            fis.read(result);
            long end = System.currentTimeMillis();
            log.info("读取文件成功............");
            log.info("读取文件耗时: {} ms", String.valueOf((end - start)));
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 在finally语句块当中确保流一定关闭。
            try {
                if (null != fis) {
                    fis.close();
                }
            } catch (Exception e2) {
                // TODO: handle exception
                e2.printStackTrace();
            }
        }
        return result;
    }

    /**
     * 通过 FileInputStream 读取文件内容（字节流），最大读取文件大小为 Integer.MAX_VALUE (B)
     * @param path 文件路径
     * @return 二进制文件内容
     */
    public static byte[] readFileByFileInputStream(String path) {
        File file = new File(path);
        log.info("读取文件路径: {}", file.getAbsolutePath());
        byte[] result = null;
        try {
            if (!file.exists()) {
                log.info("文件不存在，读取文件失败！");
                return null;
            }
            result = readFileByInputStream(new FileInputStream(file));
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }
}